from PIL import Image, ImageDraw

def create_icon():
    # 创建一个 256x256 的图像
    size = 256
    image = Image.new('RGBA', (size, size), (255, 255, 255, 0))
    draw = ImageDraw.Draw(image)
    
    # 设置颜色
    primary_color = '#2196F3'    # 蓝色
    secondary_color = '#4CAF50'  # 绿色
    arrow_color = '#FF5722'      # 橙色
    
    # 绘制圆形背景
    padding = 20
    draw.ellipse([padding, padding, size-padding, size-padding], 
                 fill='#FFFFFF', outline=primary_color, width=8)
    
    # 绘制传输图标
    # 左边的文件图标
    file_box_size = 80
    file_x = 70
    file_y = 88
    draw.rectangle([file_x, file_y, file_x+file_box_size, file_y+file_box_size], 
                  fill=primary_color)
    
    # 右边的文件图标
    file2_x = size - file_x - file_box_size
    file2_y = file_y
    draw.rectangle([file2_x, file2_y, file2_x+file_box_size, file2_y+file_box_size], 
                  fill=secondary_color)
    
    # 绘制箭头
    arrow_points = [
        (file_x + file_box_size + 10, file_y + file_box_size//2),  # 起点
        (file2_x - 10, file_y + file_box_size//2),                 # 终点
        (file2_x - 30, file_y + file_box_size//2 - 20),           # 上箭头
        (file2_x - 10, file_y + file_box_size//2),                # 回到终点
        (file2_x - 30, file_y + file_box_size//2 + 20),           # 下箭头
    ]
    draw.line(arrow_points, fill=arrow_color, width=8)
    
    # 保存为ICO文件
    image.save('file_transfer.ico', format='ICO', sizes=[(256, 256)])

if __name__ == '__main__':
    create_icon() 